package com.fragrance.util;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.CellRangeAddress;
import org.apache.poi.hssf.util.HSSFColor;

public class ExportExcel {

	/*
	 * 导出数据
	 */
	public static byte[] export(List<Object[]> dataList, String title, String[] rowName) throws Exception {
		ByteArrayOutputStream out = null;
		try {
			HSSFWorkbook workbook = new HSSFWorkbook(); // 创建工作簿对象
			HSSFSheet sheet = workbook.createSheet(title); // 创建工作表
			// 产生表格标题行
			HSSFRow rowm = sheet.createRow(0);
			HSSFCell cellTiltle = rowm.createCell(0);

			// sheet样式定义【getColumnTopStyle()/getStyle()均为自定义方法 - 在下面 - 可扩展】
			HSSFCellStyle columnTopStyle = getColumnTopStyle(workbook);// 获取列头样式对象
			HSSFCellStyle style = getStyle(workbook); // 单元格样式对象
			/*
			 * sheet.addMergedRegion(new
			 * CellRangeAddress(0,dataList.get(0).length-1 , 0,
			 * (rowName.length-1)));
			 */
			// 合并单元格
			sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, dataList.get(0).length - 1));// 列行
			cellTiltle.setCellStyle(style);
			cellTiltle.setCellValue(title);

			// 定义所需列数
			int columnNum = rowName.length;
			HSSFRow rowRowName = sheet.createRow(2); // 在索引2的位置创建行(最顶端的行开始的第二行)

			// 将列头设置到sheet的单元格中
			for (int n = 0; n < columnNum; n++) {
				HSSFCell cellRowName = rowRowName.createCell(n); // 创建列头对应个数的单元格
				cellRowName.setCellType(HSSFCell.CELL_TYPE_STRING); // 设置列头单元格的数据类型
				HSSFRichTextString text = new HSSFRichTextString(rowName[n]);
				cellRowName.setCellValue(text); // 设置列头单元格的值
				cellRowName.setCellStyle(columnTopStyle); // 设置列头单元格样式
			}

			// 将查询出的数据设置到sheet对应的单元格中
			for (int i = 0; i < dataList.size(); i++) {

				Object[] obj = dataList.get(i);// 遍历每个对象
				HSSFRow row = sheet.createRow(i + 3);// 创建所需的行数（从第三行开始写数据）

				for (int j = 0; j < obj.length; j++) {
					HSSFCell cell = null; // 设置单元格的数据类型
					// if (j == 0) {
					// cell = row.createCell(j, HSSFCell.CELL_TYPE_NUMERIC);
					// cell.setCellValue(i);
					// } else {
					cell = row.createCell(j, HSSFCell.CELL_TYPE_STRING);
					if (obj[j] != null) {
						cell.setCellValue(obj[j].toString());
					}
					// }
					cell.setCellStyle(style); // 设置单元格样式
				}
			}
			// 让列宽随着导出的列长自动适应
			for (int colNum = 0; colNum < dataList.get(1).length; colNum++) {
				int columnWidth = sheet.getColumnWidth(colNum) / 256;
				for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
					HSSFRow currentRow;
					// 当前行未被使用过
					if (sheet.getRow(rowNum) == null) {
						currentRow = sheet.createRow(rowNum);
					} else {
						currentRow = sheet.getRow(rowNum);
					}
					/*
					 * if (currentRow.getCell(colNum) != null) { HSSFCell
					 * currentCell = currentRow.getCell(colNum); if
					 * (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING)
					 * { int length =
					 * currentCell.getStringCellValue().getBytes().length; if
					 * (columnWidth < length) { columnWidth = length; } } }
					 */
					if (currentRow.getCell(colNum) != null) {
						HSSFCell currentCell = currentRow.getCell(colNum);
						if (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING) {
							int length = 0;
							try {
								length = currentCell.getStringCellValue().getBytes().length;
							} catch (Exception e) {
								e.printStackTrace();
							}
							if (columnWidth < length) {
								columnWidth = length;
							}
						}
					}

				}
				if (colNum == 0) {
					sheet.setColumnWidth(colNum, (columnWidth - 2) * 256);
				} else {
					sheet.setColumnWidth(colNum, (columnWidth + 4) * 256);
				}
			}
			if (workbook != null) {
				try {
					out = new ByteArrayOutputStream();
					workbook.write(out);
					return out.toByteArray();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}

		} catch (Exception e) {
			throw e;
		} finally {
			if (out != null) {
				out.close();
			}
		}
		return null;
	}

	/*
	 * 列头单元格样式
	 */
	public static HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {

		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		font.setFontHeightInPoints((short) 11);
		// 字体加粗
		font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		// 设置底边框颜色;
		style.setBottomBorderColor(HSSFColor.BLACK.index);
		// 设置左边框;
		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		// 设置左边框颜色;
		style.setLeftBorderColor(HSSFColor.BLACK.index);
		// 设置右边框;
		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		// 设置右边框颜色;
		style.setRightBorderColor(HSSFColor.BLACK.index);
		// 设置顶边框;
		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		// 设置顶边框颜色;
		style.setTopBorderColor(HSSFColor.BLACK.index);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		// 设置垂直对齐的样式为居中对齐;
		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		return style;

	}

	/*
	 * 列数据信息单元格样式
	 */
	public static HSSFCellStyle getStyle(HSSFWorkbook workbook) {
		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		// font.setFontHeightInPoints((short)10);
		// 字体加粗
		// font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		// 设置底边框颜色;
		style.setBottomBorderColor(HSSFColor.BLACK.index);
		// 设置左边框;
		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		// 设置左边框颜色;
		style.setLeftBorderColor(HSSFColor.BLACK.index);
		// 设置右边框;
		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		// 设置右边框颜色;
		style.setRightBorderColor(HSSFColor.BLACK.index);
		// 设置顶边框;
		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		// 设置顶边框颜色;
		style.setTopBorderColor(HSSFColor.BLACK.index);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		// 设置垂直对齐的样式为居中对齐;
		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
		return style;

	}

	public static List<Object[]> readXls(File xlsFile) throws Exception {
		if (!xlsFile.exists()) {
			throw new Exception("文件不存在");
		}
		FileInputStream in = null;
		HSSFSheet sheet = null;
		List<Object[]> result = new ArrayList<Object[]>();
		try {
			in = new FileInputStream(xlsFile);
			HSSFWorkbook workbook = new HSSFWorkbook(in);
			// 获取每个Sheet表
			for (int i = 0; i < workbook.getNumberOfSheets(); i++) {
				sheet = workbook.getSheetAt(i);
				// getLastRowNum，获取最后一行的行标
				for (int j = 0; j < sheet.getLastRowNum(); j++) {
					HSSFRow row = sheet.getRow(j);
					if (row != null) {
						Object[] rowObj = new Object[row.getLastCellNum()];
						// getLastCellNum，是获取最后一个不为空的列是第几个
						for (int k = 0; k < row.getLastCellNum(); k++) {
							// getCell 获取单元格数据
							if (row.getCell(k) != null) {
								rowObj[k] = row.getCell(k);
							}
						}
						result.add(rowObj);
					}
				}
			}
			return result;
		} catch (Exception e) {
			throw e;
		} finally {
			if (in != null) {
				try {
					in.close();
				} catch (Exception e) {
				}
			}
		}
	}

	public static List<Object[]> readXls(String filePath) throws Exception {
		if (StringUtils.isEmpty(filePath)) {
			throw new Exception("filePath不能为空");
		}
		return readXls(new File(filePath));
	}

	public static void main(String[] args) throws Exception {
		List<Object[]> dataList = new ArrayList<Object[]>();
		Object[] d1 = new Object[2];
		d1[0] = 1000;
		d1[1] = 100;
		dataList.add(d1);
		Object[] d2 = new Object[2];
		d2[0] = 2000;
		d2[1] = 200;
		dataList.add(d2);
		Object[] d3 = new Object[2];
		d3[0] = 3000;
		d3[1] = 300;
		dataList.add(d3);
		byte[] b = ExportExcel.export(dataList,"财报", new String[] { "营业额", "利润" });
		if (b != null) {
			FileOutputStream out = new FileOutputStream(new File("d:/a.xls"));
			out.write(b);
		}
		
		ExportExcel.readXls("d:/1.xls");
		System.out.println("--------");
	}

}